package org.jsola.hr.service.impl;

import org.jsola.core.Page;
import org.jsola.hr.common.ExampleKit;
import org.jsola.hr.constant.HrConstants;
import org.jsola.hr.dao.ICompanySaleUserDAO;
import org.jsola.hr.dto.CompanySaleUserAddDTO;
import org.jsola.hr.dto.CompanySaleUserUpdateDTO;
import org.jsola.hr.entity.CompanyDO;
import org.jsola.hr.entity.CompanySaleUserDO;
import org.jsola.hr.entity.SocProgItemDO;
import org.jsola.hr.query.CompanySaleUserQuery;
import org.jsola.hr.service.ICompanySaleUserService;
import org.jsola.hr.service.ICompanyService;
import org.jsola.hr.vo.CompanySaleUserListVO;
import org.jsola.hr.vo.CompanySaleUserVO;
import org.jsola.hr.vo.CompanyVO;
import org.jsola.user.core.TokenUser;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Service;
import org.springframework.transaction.annotation.Transactional;
import org.springframework.util.CollectionUtils;
import tk.mybatis.mapper.entity.Example;

import java.util.ArrayList;
import java.util.Comparator;
import java.util.List;
import java.util.Map;
import java.util.stream.Collectors;

/**
 * 用工单位销售
 *
 * @author zhr
 */
@Service("hrCompanySaleUserServiceImpl")
public class CompanySaleUserServiceImpl implements ICompanySaleUserService {

    @Autowired
    private ICompanySaleUserDAO companySaleUserDAO;

    @Autowired
    private ICompanyService companyService;

    @Override
    @Transactional(value = HrConstants.TX, rollbackFor = Exception.class)
    public CompanySaleUserVO save(CompanySaleUserAddDTO companySaleUserAddDTO, TokenUser tokenUser) {
        // 转DO
        CompanySaleUserDO companySaleUserDO = companySaleUserAddDTO.to(CompanySaleUserDO.class);
        // 保存
        companySaleUserDO = save(companySaleUserDO, tokenUser.getSiteId(), tokenUser.getUserId());
        return companySaleUserDO.to(CompanySaleUserVO.class);
    }

    @Override
    @Transactional(value = HrConstants.TX, rollbackFor = Exception.class)
    public int update(CompanySaleUserUpdateDTO companySaleUserUpdateDTO, TokenUser tokenUser) {
        //转DO
        CompanySaleUserDO companySaleUserDO = companySaleUserUpdateDTO.to(CompanySaleUserDO.class);
        //根据主键更新，只更新非null值
        return updateByIdSelective(companySaleUserDO, tokenUser.getSiteId(), tokenUser.getUserId());
    }

    @Override
    @Transactional(value = HrConstants.TX, rollbackFor = Exception.class)
    public int deleteByIds(TokenUser tokenUser, String... companySaleUserIds) {
        return logicalDeleteByIds(tokenUser.getSiteId(), tokenUser.getUserId(), (Object[]) companySaleUserIds);
    }


    @Override
    public CompanySaleUserVO selectById(String companySaleUserId, String siteId) {
        CompanySaleUserDO companySaleUserDO = selectDOById(companySaleUserId, siteId);
        if (companySaleUserDO == null) {
            return null;
        }
        return companySaleUserDO.to(CompanySaleUserVO.class);
    }

    @Override
    public List<CompanySaleUserListVO> select(CompanySaleUserQuery companySaleUserQuery, String siteId) {
        List<CompanySaleUserDO> companySaleUserDOList = selectDO(companySaleUserQuery, siteId);
        if (CollectionUtils.isEmpty(companySaleUserDOList)) {
            return companySaleUserDOList == null ? null : new ArrayList<>();
        }
        return companySaleUserDOList.stream()
                .map(companySaleUserDO -> companySaleUserDO.to(CompanySaleUserListVO.class))
                .collect(Collectors.toList());
    }

    @Override
    public int selectCount(CompanySaleUserQuery companySaleUserQuery, String siteId) {
        Example example = buildExample(companySaleUserQuery, siteId);
        return companySaleUserDAO.selectCountByExample(example);
    }

    @Override
    public Page<CompanySaleUserListVO> selectPage(CompanySaleUserQuery companySaleUserQuery, String siteId) {
        Example example = buildExample(companySaleUserQuery, siteId);
        Page<CompanySaleUserDO> page = companySaleUserDAO.selectPageByExample(example,
                companySaleUserQuery.getPageNo(),
                companySaleUserQuery.getPageSize());

        return page.to(CompanySaleUserListVO.class);
    }

    @Override
    public CompanySaleUserDO selectDOById(String companySaleUserId, String siteId) {
        return listById(companySaleUserId, siteId);
    }

    @Override
    public List<CompanySaleUserDO> selectDO(CompanySaleUserQuery companySaleUserQuery, String siteId) {
        Example example = buildExample(companySaleUserQuery, siteId);
        return companySaleUserDAO.selectByExample(example);
    }

    @Override
    public void bachSave(List<CompanySaleUserAddDTO> saleUserList, TokenUser tokenUser) {
        if (!CollectionUtils.isEmpty(saleUserList)) {
            List<CompanySaleUserDO> insertList = saleUserList
                    .parallelStream()
                    .map(companySaleUser -> {
                        CompanySaleUserDO saleUser = companySaleUser.to(CompanySaleUserDO.class);
                        saleUser.preInsert(tokenUser.getUserId());
                        saleUser.setSiteId(tokenUser.getSiteId());
                        return saleUser;
                    }).collect(Collectors.toList());
            companySaleUserDAO.insertListAndSetId(insertList);
        }
    }

    @Override
    public List<CompanySaleUserDO> selectSaleUserByCompanyId(String companyId, String siteId) {
        Example example = new Example(CompanySaleUserDO.class);
        example.and()
                .andEqualTo("companyId", companyId)
                .andEqualTo("valid", true)
                .andEqualTo("siteId", siteId);
        return companySaleUserDAO.selectByExample(example);
    }

    @Override
    public void bachSave(List<CompanySaleUserDO> saveList) {
        companySaleUserDAO.insertListAndSetId(saveList);
    }

    @Override
    public void deleteFromCompany(String companyId, TokenUser tokenUser) {
        CompanySaleUserDO updateTemplate = new CompanySaleUserDO();
        updateTemplate.setValid(false);
        Example example = new Example(CompanySaleUserDO.class);
        example.and()
                .andEqualTo("companyId", companyId)
                .andEqualTo("valid", true)
                .andEqualTo("siteId", tokenUser.getSiteId());
        companySaleUserDAO.updateByExampleSelective(updateTemplate, example);
    }

    @Override
    public List<CompanyVO> selectCompanyBySaleUserId(String userId, TokenUser tokenUser) {
        Example example = new Example(CompanySaleUserDO.class);
        example.and()
                .andEqualTo("valid", true)
                .andEqualTo("saleUserId", userId)
                .andEqualTo("siteId", tokenUser.getSiteId());
        List<CompanySaleUserDO> companySaleUser = companySaleUserDAO.selectByExample(example);
        if(!CollectionUtils.isEmpty(companySaleUser)){
            List<String> companyIdList = companySaleUser.stream()
                    .map(CompanySaleUserDO::getCompanyId).collect(Collectors.toList());
            Map<String, CompanySaleUserDO> companySaleUserMap = companySaleUser.stream()
                    .collect(Collectors.toMap(CompanySaleUserDO::getCompanyId, v -> v));
            List<CompanyDO> companyList = companyService.selectByIds(companyIdList, tokenUser.getSiteId());
            return companyList.stream().map(company->{
                CompanySaleUserDO saleUser = companySaleUserMap.get(company.getId());
                CompanyVO toResult = company.to(CompanyVO.class);
                toResult.setSaleBindTime(saleUser.getGmtCreate());
                return toResult;
            }).sorted(Comparator.comparing(CompanyVO::getSaleBindTime).reversed()).collect(Collectors.toList());
        }
        return new ArrayList<>();
    }

    /**
     * 根据查询参数，构建example
     *
     * @param companySaleUserQuery 查询参数
     * @param siteId               所属站点id
     * @return example
     */
    private Example buildExample(CompanySaleUserQuery companySaleUserQuery, String siteId) {
        Example example = new Example(CompanySaleUserDO.class);
        example.and()
                .andEqualTo("valid", true)
                .andEqualTo("siteId", siteId);
        // 排序
        ExampleKit.setExampleOrder(example, companySaleUserQuery.getOrders());
        return example;
    }
}




